package com.hnchances.studentscoresystem.util;

import jakarta.servlet.http.HttpServletRequest;
import lombok.extern.slf4j.Slf4j;



/**
 * IP工具类
 * 功能：获取客户端真实IP地址（适配代理、负载均衡、Nginx转发等场景）
 * 对应文档需求：user表loginIP字段存储用户登录IPv4地址
 */
@Slf4j
public class IpUtil {

    /**
     * 常见的代理IP请求头（优先从这些头中获取真实IP）
     */
    private static final String[] PROXY_HEADERS = {
            "X-Forwarded-For",
            "Proxy-Client-IP",
            "WL-Proxy-Client-IP",
            "HTTP_X_FORWARDED_FOR",
            "HTTP_X_FORWARDED",
            "HTTP_X_CLUSTER_CLIENT_IP",
            "HTTP_CLIENT_IP",
            "HTTP_FORWARDED_FOR",
            "HTTP_FORWARDED",
            "X-Real-IP"
    };

    /**
     * 本地IP地址（排除本地回环地址）
     */
    private static final String LOCAL_IP = "127.0.0.1";
    private static final String LOCAL_IPV6 = "0:0:0:0:0:0:0:1";

    /**
     * 获取客户端真实IP地址
     * @param request HttpServletRequest对象
     * @return 客户端真实IP（IPv4格式，如192.168.1.1）
     */
    public static String getClientIp(HttpServletRequest request) {
        if (request == null) {
            log.warn("IP获取失败：HttpServletRequest对象为null");
            return LOCAL_IP;
        }

        // 1. 优先从代理请求头中获取IP（适配代理场景）
        for (String header : PROXY_HEADERS) {
            String ip = request.getHeader(header);
            if (isValidIp(ip)) {
                log.debug("从请求头[{}]获取到客户端IP: {}", header, ip);
                return ip;
            }
        }

        // 2. 若代理头中无有效IP，从request.getRemoteAddr()获取（直接访问场景）
        String remoteIp = request.getRemoteAddr();
        log.debug("从RemoteAddr获取到客户端IP: {}", remoteIp);
        // 处理IPv6本地地址，转为IPv4本地地址
        if (LOCAL_IPV6.equals(remoteIp)) {
            remoteIp = LOCAL_IP;
        }
        return remoteIp;
    }

    /**
     * 校验IP地址是否有效（非空、非unknown、非本地回环地址）
     * @param ip 待校验IP地址
     * @return true-有效IP，false-无效IP
     */
    private static boolean isValidIp(String ip) {
        if (ip == null || ip.trim().isEmpty() || "unknown".equalsIgnoreCase(ip.trim())) {
            return false;
        }
        // 排除本地回环地址（仅在代理场景下，若代理头中为本地IP，视为无效）
        String trimmedIp = ip.trim();
        if (LOCAL_IP.equals(trimmedIp) || LOCAL_IPV6.equals(trimmedIp)) {
            return false;
        }
        // 若为多IP（如X-Forwarded-For: 192.168.1.1, 10.0.0.1），取第一个非unknown的IP
        if (trimmedIp.contains(",")) {
            String[] ipArray = trimmedIp.split(",");
            for (String singleIp : ipArray) {
                if (isValidSingleIp(singleIp.trim())) {
                    return true;
                }
            }
            return false;
        }
        // 校验单个IP格式（简化校验，仅判断是否符合IPv4格式的大致结构）
        return isValidSingleIp(trimmedIp);
    }

    /**
     * 校验单个IP地址是否符合IPv4格式（简化版，仅判断是否为xxx.xxx.xxx.xxx结构）
     * @param ip 待校验单个IP地址
     * @return true-符合IPv4格式，false-不符合
     */
    private static boolean isValidSingleIp(String ip) {
        if (ip == null || ip.trim().isEmpty() || "unknown".equalsIgnoreCase(ip)) {
            return false;
        }
        // IPv4格式正则（简化版，不校验每个段的数值范围，仅校验结构）
        String ipv4Pattern = "^(\\d{1,3}\\.){3}\\d{1,3}$";
        return ip.matches(ipv4Pattern);
    }
}